SpringMvcRequestExecutor.java

package org.restler.spring.mvc.spring;

import com.google.common.collect.ImmutableMultimap;
import org.restler.client.Call;
import org.restler.client.RestlerException;
import org.restler.http.*;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpMethod;
import org.springframework.http.RequestEntity;
import org.springframework.http.ResponseEntity;
import org.springframework.http.converter.HttpMessageNotWritableException;
import org.springframework.util.MultiValueMap;
import org.springframework.web.client.HttpStatusCodeException;
import org.springframework.web.client.RestTemplate;

public class SpringMvcRequestExecutor implements RequestExecutor {

    private final RestTemplate restTemplate;

    public SpringMvcRequestExecutor(RestTemplate restTemplate) {
        this.restTemplate = restTemplate;
    }

    public Response execute(Call call) {
        HttpCall springMvcCall = SpringUtils.prepareForSpringMvc((HttpCall) call);
        RequestEntity<?> requestEntity = toRequestEntity(springMvcCall);
        ResponseEntity responseEntity;
        try {
            responseEntity = restTemplate.exchange(requestEntity, new HttpCallTypeReference<>(call));
        } catch (HttpStatusCodeException e) {
            HttpStatus status = new HttpStatus(e.getStatusCode().value(), e.getStatusText());
            return new FailedResponse(status, e.getResponseBodyAsString(), e);
        } catch (HttpMessageNotWritableException e) {
            throw new RestlerException("Could not serialize body", e);
        }

        ImmutableMultimap<String, String> headersBuilder = SpringUtils.toGuavaMultimap(responseEntity.getHeaders());
        HttpStatus status = new HttpStatus(responseEntity.getStatusCode().value(), responseEntity.getStatusCode().getReasonPhrase());

        return new SuccessfulResponse(status, headersBuilder, responseEntity.getBody());
    }

    private RequestEntity<?> toRequestEntity(HttpCall call) {
        MultiValueMap<String, String> headers = new HttpHeaders();
        call.getHeaders().entries().stream().forEach(entry -> headers.add(entry.getKey(), entry.getValue()));
        return new RequestEntity<>(call.getRequestBody(), headers, HttpMethod.valueOf(call.getHttpMethod().name()), call.getUrl());
    }

}